import gradio as gr

from iflytech_assistant.assistant.girl_code import generate

iface = gr.Interface(
    fn=generate,
    inputs=gr.Textbox(lines=1, label="女生话术"),
    outputs=[gr.TextArea(label="Prompt"), gr.Textbox(lines=2, label="女生暗语")],
    title="女生暗语生成",
    description="输入女生话术，生成女生暗语。",
)

iface.launch(share=True)
